package com.fhzn.auth.mapper;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.fhzn.auth.entity.UserRelation;
import org.apache.ibatis.annotations.Mapper;

import java.util.List;

/**
 * @author 老顾
 * @apiNote
 * @since 2023/12/26
 */
@Mapper
public interface UserRelationMapper extends BaseMapper<UserRelation> {

    /**
     * 查询用户和角色的关系
     *
     * @param id      用户id
     * @param userIds 用户id列表
     * @return 用户关系列表
     */
    default List<UserRelation> findRelationsWithUsers(Integer id, List<Integer> userIds) {
        LambdaQueryWrapper<UserRelation> wrapper = Wrappers.lambdaQuery();
        wrapper.eq(UserRelation::getLeaderId, id);
        wrapper.in(UserRelation::getUserId, userIds);
        return selectList(wrapper);
    }

    /**
     * 查询用户和角色的关系
     *
     * @param leaderId 上级id
     * @param userId   用户id
     * @return 用户关系列表
     */
    default List<UserRelation> findRelations(Integer leaderId, Integer userId) {
        LambdaQueryWrapper<UserRelation> wrapper = Wrappers.lambdaQuery();
        wrapper.eq(UserRelation::getLeaderId, leaderId);
        wrapper.eq(UserRelation::getUserId, userId);
        return selectList(wrapper);
    }

}
